import argparse


def parse_args():
    parser = argparse.ArgumentParser(description="Parameters of preprocessing and statistical program")
    parser.add_argument("--root-dir", default="/home/shaohua/data2/Datasets/Face_Anti_Spoofing/Oulu_NPU", help="Root directory")
    parser.add_argument("--protocol", type=str, default="Protocol_1",
                        choices=["Protocol_1", "Protocol_2", "Protocol_3", "Protocol_4"], help="Manipulation type")
    parser.add_argument("--detector-type", default="FacenetDetector", choices=["FacenetDetector"],
                        help="type of the detector")
    parser.add_argument("--crops-dir", default="crops", help="crops directory")
    args = parser.parse_args()
    return args
